package 链表题目;

import 链表题目.model.ListNode;


//两数相加
//https://leetcode.cn/problems/add-two-numbers/description/
class 两数相加 {
    // 我的解法
    public ListNode addTwoNumbers1(ListNode l1, ListNode l2) {
        int addNum = 0;//记录进位
        ListNode ans = new ListNode(0, null);
        ListNode cur = ans;
        while (l1 != null && l2 != null) {
            int sum = l1.val + l2.val + addNum;
            addNum = sum / 10;//记录进位
            cur.next = new ListNode(sum % 10, null);
            cur = cur.next;
            l1 = l1.next;
            l2 = l2.next;
        }
        // 同时遍历到指针尾部处理未添加的进位
        if (l1 == null && l2 == null && addNum != 0) {
            cur.next = new ListNode(addNum, null);
        }
        // 后续进位操作判断是否有进位情况，如果有通过dfs进行进位操作
        if (l1 != null) {
            dfs(l1, addNum);
            cur.next = l1;
        }
        if (l2 != null) {
            dfs(l2, addNum);
            cur.next = l2;
        }
        return ans.next;
    }

    public static void dfs(ListNode head, int addNum) {
        // 计算两数之和
        int sum = head.val + addNum;
        // 进位情况
        addNum = sum / 10;
        // 当前节点取值
        head.val = sum % 10;
        if (head.next == null) {
            // 如果下一个节点为null，且有进位未处理 则进行进位
            if (addNum != 0) {
                head.next = new ListNode(addNum, null);
            }
            return;
        }
        if (sum >= 10) {
            dfs(head.next, addNum);
        }
    }

    // 优化后-> 提交之后没什么区别 我的运行内存43.7MB
    public ListNode addTwoNumbers2(ListNode l1, ListNode l2) {
        int addNum = 0;//记录进位
        ListNode ans = new ListNode(0, l1);
        ListNode temp = null;
        while (l1 != null && l2 != null) {
            int sum = l1.val + l2.val + addNum;
            addNum = sum / 10;//记录进位
            l1.val = sum % 10;
            temp = l1;
            l1 = l1.next;
            l2 = l2.next;
        }
        // 同时遍历到指针尾部处理未添加的进位
        if (l1 == null && l2 == null && addNum != 0) {
            temp.next = new ListNode(addNum, null);
        }
        // 后续进位操作判断是否有进位情况，如果有通过dfs进行进位操作
        if (l1 != null) {
            dfs(l1, addNum);
        }
        if (l2 != null) {
            dfs(l2, addNum);
            temp.next = l2;
        }
        return ans.next;
    }

    // 官方题解 官方运行内存43.4MB 没啥区别的样子
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode head = null, tail = null;
        int carry = 0;
        while (l1 != null || l2 != null) {
            int n1 = l1 != null ? l1.val : 0;
            int n2 = l2 != null ? l2.val : 0;
            int sum = n1 + n2 + carry;
            if (head == null) {
                head = tail = new ListNode(sum % 10);
            } else {
                tail.next = new ListNode(sum % 10);
                tail = tail.next;
            }
            carry = sum / 10;
            if (l1 != null) {
                l1 = l1.next;
            }
            if (l2 != null) {
                l2 = l2.next;
            }
        }
        if (carry > 0) {
            tail.next = new ListNode(carry);
        }
        return head;
    }

//    public static void main(String[] args) {
//        ListNode tow5 = new ListNode(9, null);
//        ListNode tow4 = new ListNode(6, tow5);
//        ListNode tow3 = new ListNode(4, tow4);
//        ListNode tow2 = new ListNode(6, tow3);
//        ListNode tow1 = new ListNode(5, null);
//        ListNode one4 = new ListNode(4, null);
//        ListNode one3 = new ListNode(3, one4);
//        ListNode one2 = new ListNode(4, one3);
//        ListNode one1 = new ListNode(5, null);
//        new AddTwoNumbers().addTwoNumbers(one1, tow1);
//    }
}
